#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
数据分析师 Agent
负责分析股票数据
"""

from crewai import Agent
from prompts import get_data_analyst_prompt


def create_data_analyst_agent(llm=None) -> Agent:
    """
    创建数据分析师智能体
    
    Args:
        llm: 语言模型实例（可选）
    
    Returns:
        配置好的数据分析师 Agent
    """
    agent_config = {
        "role": "资深股票分析师",
        "goal": "对股票进行全面的技术分析和基本面分析，给出专业的投资建议",
        "backstory": get_data_analyst_prompt(),
        "verbose": True,
        "allow_delegation": False
    }
    
    if llm is not None:
        agent_config["llm"] = llm
    
    return Agent(**agent_config)
